/*
 * The contents of this file are subject to the Mozilla Public License
 * Version 1.1 (the "License"); you may not use this file except in
 * compliance with the License. You may obtain a copy of the License at
 * http://www.mozilla.org/MPL/
 *
 * Software distributed under the License is distributed on an "AS IS"
 * basis, WITHOUT WARRANTY OF ANY KIND, either express or implied. See the
 * License for the specific language governing rights and limitations
 * under the License.
 * 
 * The Original Code is iSQL-Viewer, A Mutli-Platform Database Tool.
 *
 * The Initial Developer of the Original Code is iSQL-Viewer, A Mutli-Platform Database Tool.
 * Portions created by Mark A. Kobold are Copyright (C) 2000-2007. All Rights Reserved.
 *
 * Contributor(s): 
 *  Mark A. Kobold [mkobold <at> isqlviewer <dot> com].
 *  
 * If you didn't download this code from the following link, you should check
 * if you aren't using an obsolete version: http://www.isqlviewer.com
 */
package org.isqlviewer.ui.laf;

import java.awt.Color;
import java.awt.Graphics;
import java.awt.Insets;
import java.awt.Rectangle;

import javax.swing.UIManager;
import javax.swing.plaf.basic.BasicTreeUI;
import javax.swing.tree.TreePath;

import org.isqlviewer.swing.outline.JOutline;

/**
 * This TreeUI does some nicer things when it comes to dealing with the JOutline (JTreeTable) making it appear as a
 * single component.
 * <p>
 * This UI mainly addresses selection issues at this time where there is a distinct visual disconnect between the tree
 * being rendered within the table and the table itself.
 * 
 * @author Mark A. Kobold &lt;mkobold at isqlviewer dot com&gt;
 * @version 1.0
 */
public class EnhancedTreeUI extends BasicTreeUI {

    private JOutline outline = null;

    public EnhancedTreeUI(JOutline outline) {

        this.outline = outline;
        setExpandedIcon(UIManager.getIcon("Tree.expandedIcon"));
        setCollapsedIcon(UIManager.getIcon("Tree.collapsedIcon"));
    }

    /**
     * Paints the renderer part of a row. The receiver should NOT modify <code>clipBounds</code>, or
     * <code>insets</code>.
     */
    @Override
    protected void paintRow(Graphics g, Rectangle clipBounds, Insets insets, Rectangle bounds, TreePath path, int row,
            boolean isExpanded, boolean hasBeenExpanded, boolean isLeaf) {

        boolean isSelected = false;
        for (int selection : outline.getSelectedRows()) {
            if (selection == row) {
                isSelected = true;
                break;
            }
        }

        if (isSelected) {
            if (outline.hasFocus()) {
                g.setColor(EnhancedTableCellRenderer.selectedFocusedColor);
                g.fillRect(clipBounds.x, clipBounds.y, clipBounds.width, clipBounds.height);
            } else {
                g.setColor(EnhancedTableCellRenderer.selectedNotFocusedColor);
                g.fillRect(clipBounds.x, clipBounds.y, clipBounds.width, clipBounds.height);
            }
        } else {
            Color color = row % 2 == 0 ? EnhancedTableCellRenderer.evenRowColor : EnhancedTableCellRenderer.oddRowColor;
            g.setColor(color);
            g.fillRect(clipBounds.x, clipBounds.y, clipBounds.width, clipBounds.height);
        }
        if (shouldPaintExpandControl(path, row, isExpanded, hasBeenExpanded, isLeaf)) {
            paintExpandControl(g, clipBounds, insets, bounds, path, row, isExpanded, hasBeenExpanded, isLeaf);
        }
        super.paintRow(g, clipBounds, insets, bounds, path, row, isExpanded, hasBeenExpanded, isLeaf);
    }
}
